package sliding_window

/*
904.你正在探访一家农场，农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示，其中 fruits[i] 是第 i 棵树上的水果【种类】  。

你想要尽可能多地收集水果。然而，农场的主人设定了一些严格的规矩，你必须按照要求采摘水果：
1.你只有【两个】篮子，并且每个篮子只能装【单一类型】的水果。每个篮子能够装的水果总量没有限制。
2.你可以选择任意一棵树开始采摘，你必须从【每棵】树（包括开始采摘的树）上【恰好摘一个水果】。采摘的水果应当符合篮子中的水果类型。每采摘一次，你将会向右移动到下一棵树，并继续采摘。
3.一旦你走到某棵树前，但水果不符合篮子的水果类型，那么就必须停止采摘。

给你一个整数数组 fruits ，返回你可以收集的水果的【最大】数目。

示例 1：
输入：fruits = [1,2,1]
输出：3
解释：可以采摘全部 3 棵树。

示例 2：
输入：fruits = [0,1,2,2]
输出：3
解释：可以采摘 [1,2,2] 这三棵树。
如果从第一棵树开始采摘，则只能采摘 [0,1] 这两棵树。

示例 3：
输入：fruits = [1,2,3,2,2]
输出：4
解释：可以采摘 [2,3,2,2] 这四棵树。
如果从第一棵树开始采摘，则只能采摘 [1,2] 这两棵树。

示例 4：
输入：fruits = [3,3,3,1,2,1,1,2,3,3,4]
输出：5
解释：可以采摘 [1,2,1,1,2] 这五棵树。
*/

//问题转换为找到最长的子序列，最多含有两种类型
//时间复杂度：O(N)
//空间复杂度：O(N)
func totalFruit(fruits []int) int {
	// window的key可以区分类型
	ans, left, right, window := 0, 0, 0, make(map[int]int)

	// expands window
	for right < len(fruits) {
		c := fruits[right]
		right++
		window[c]++

		//shrink window
		//只有两个篮子，也就是说map的大小为小于等于2
		for len(window) > 2 {
			d := fruits[left]
			left++
			window[d]--
			//篮子里没有水果删除
			if window[d] == 0 {
				delete(window, d)
			}

		}

		//只能恰好摘一个水果
		ans = max(ans, right-left)
	}

	return ans
}
